square root a number in python 3